package com.vic.delayedqueue.config;

import org.springframework.amqp.core.*;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.HashMap;
import java.util.Map;

/**
 * rabbitmq的延时队列配置
 * 1 声明正常交换机、正常路由键、正常队列、正常队列绑定到正常交换机
 * 2 声明死信交换机、死信路由键、死信队列、死信队列绑定到死信交换机
 * 3 在声明正常队列时，指定死信交换机、死信路由键、队列的TTL
 */
@Configuration
public class DelayedQueue1Config {

    // 正常交换机
    public static final String NORMAL_EXCHANGE = "normal_exchange";
    // 正常队列1
    public static final String NORMAL_QUEUE_1 = "normal_queue_1";
    // 正常队列2
    public static final String NORMAL_QUEUE_2 = "normal_queue_2";
    // 正常路由键1
    public static final String NORMAL_ROUTING_KEY_1 = "nK1";
    // 正常路由键2
    public static final String NORMAL_ROUTING_KEY_2 = "nK2";

    // 死信交换机
    public static final String DEAD_EXCHANGE = "dead_exchange";
    // 死信队列
    public static final String DEAD_QUEUE = "dead_queue";
    // 死信路由键
    public static final String DEAD_ROUTING_KEY = "dk";

    // 声明普通交换机
    @Bean("normal_exchange")
    public DirectExchange directExchange() {
        return new DirectExchange(NORMAL_EXCHANGE, true, false);
    }

    // 声明普通队列1，当该队列中的消息超过30s未消费，会将消息传递到死信交换机
    @Bean("normal_queue_1")
    public Queue normalQueue1() {
        Map<String, Object> arguments = new HashMap<>();
        // 设置队列的死信交换机
        arguments.put("x-dead-letter-exchange", DEAD_EXCHANGE);
        // 设置死信的 routing key
        arguments.put("x-dead-letter-routing-key", DEAD_ROUTING_KEY);
        // 设置消息超时时间，单位ms
        arguments.put("x-message-ttl", 30 * 1000);
        return QueueBuilder.durable(NORMAL_QUEUE_1).withArguments(arguments).build();
    }

    // 声明 普通队列1 绑定 普通交换机
    @Bean
    public Binding normalQueue1BindingNormalExchange(@Qualifier("normal_queue_1") Queue queue, @Qualifier("normal_exchange") DirectExchange exchange) {
        return BindingBuilder.bind(queue).to(exchange).with(NORMAL_ROUTING_KEY_1);
    }


    // 声明普通队列2，当该队列中的消息超过10s未消费，会将消息传递到死信交换机
    @Bean("normal_queue_2")
    public Queue normalQueue2() {
        Map<String, Object> arguments = new HashMap<>();
        // 设置队列的死信交换机
        arguments.put("x-dead-letter-exchange", DEAD_EXCHANGE);
        // 设置死信的 routing key
        arguments.put("x-dead-letter-routing-key", DEAD_ROUTING_KEY);
        // 设置消息超时时间，单位ms
        arguments.put("x-message-ttl", 10 * 1000);
        return QueueBuilder.durable(NORMAL_QUEUE_2).withArguments(arguments).build();
    }

    // 声明 普通队列2 绑定 普通交换机
    @Bean
    public Binding normalQueue2BindingNormalExchange(@Qualifier("normal_queue_2") Queue queue, @Qualifier("normal_exchange") DirectExchange exchange) {
        return BindingBuilder.bind(queue).to(exchange).with(NORMAL_ROUTING_KEY_2);
    }

    // 声明死信交换机
    @Bean("dead_exchange")
    public DirectExchange deadExchange() {
        return new DirectExchange(DEAD_EXCHANGE, true, false);
    }

    // 声明死信队列
    @Bean("dead_queue")
    public Queue deadQueue() {
        return new Queue(DEAD_QUEUE, true, false, false);
    }

    // 声明死信队列绑定死信交换机
    @Bean
    public Binding deadQueueBindDeadExchange(@Qualifier("dead_queue") Queue queue, @Qualifier("dead_exchange") DirectExchange exchange) {
        return BindingBuilder.bind(queue).to(exchange).with(DEAD_ROUTING_KEY);
    }

}